package com.example.demo.leetcode.top;

/**
 *
 * 思路:
 *      一个定长的数组如何归并?
 *  (1) 先取出数组的中间点，根据中间点，两两拆分数组(使用递归拆分)
 *  (2) 一直拆分到数组的大小为1结束递归
 *  (3) 每次拆分数组后拿到两个新的数组，在递归中将两个数组合并排序，使用两个下标作为两个数组的起始，比较后存入临时数组。
 *      临时数组最后需要赋值给原始数组。
 */
public class Array_归并排序 {

    public static void mergeSort(int[] arr) {
        int start = 0;
        int end = arr.length - 1;
        splitSort(arr, start, end);
    }

    //进行拆分，归并
    public static void splitSort(int[] arr, int start, int end) {
        if (start < end) {
            int mid = (start + end) / 2;
            splitSort(arr, start, mid);
            splitSort(arr, mid + 1, end);
            //进行归并
            mergeTwoSort(arr, start, mid, end);
        }
    }

    //归并两个数组
    public static void mergeTwoSort(int[] arr, int start, int mid, int end) {
        //临时数组
        int[] temp = new int[end + 1];
        int i = start;
        int j = mid + 1;
        int t = 0; //；临时数组下标

        while (i <= mid && j <= end) {
            if (arr[i] < arr[j]) {
                temp[t] = arr[i];
                t++;
                i++;
            } else {
                temp[t] = arr[j];
                j++;
                t++;
            }
        }

        //兜底,如果i，j还没有到头
        while (i <= mid) {
            temp[t] = arr[i];
            t++;
            i++;
        }
        while (j <= end) {
            temp[t] = arr[j];
            t++;
            j++;
        }

        //吧temp的值赋值给 arr
        t = 0;
        int left = start;
        while (left <= end) {
            arr[left] = temp[t];
            left++;
            t++;
        }
    }

    public static void main(String[] args) {
        int[] arr = {3, 4, 6, 2, 1, 10};
        sort2(arr);
        for (int i = 0; i < arr.length; i++) {
            System.out.print(arr[i] + " ");
        }
    }

    //复习归并
    public static void sort2(int[] arr) {

        splitSort2(arr, 0, arr.length - 1);
    }

    public static void splitSort2(int[] arr, int i, int j) {

        if (i < j) {
            int middle = (i + j) / 2;
            splitSort2(arr, i, middle);
            splitSort2(arr, middle + 1, j);
            mergeSort2(arr, i, middle, j);
        }

    }

    public static void mergeSort2(int[] arr, int i, int middle, int j) {

        int[] temp = new int[j - i + 1];
        int middleEnd = middle;
        int start = i;
        int end = j;
        middle = middle + 1;
        int k = 0;
        while (i <= middleEnd &&  middle <= j) {
            if (arr[i] > arr[middle]) {
                temp[k] = arr[middle];
                middle++;
                k++;
            } else {
                temp[k] = arr[i];
                i++;
                k++;
            }
        }

        while (i <= middleEnd) {
            temp[k++] = arr[i++];
        }

        while (middle <= j) {
            temp[k++] = arr[middle++];
        }

        int count = 0;
        for (int x = start; x <= end; x++) {
            arr[x] = temp[count++];
        }

    }

}
